# 1、导入模块
import os
import paddlehub as hub

# 2、加载模型

seg = hub.Module(name='humanseg_server')

# 3、获取文件列表
# 图片文件的目录
path = 'E:/BaiduNetdiskDownload/Market1501/bounding_box_train/'
path = 'E:/datasets/real/'
out = 'E:/remove/'
# 获取目录下的文件
files = os.listdir(path)
print(files)
# 用来装图片的
imgs = []
# 拼接图片路径
for i in files:
    imgs.append(path + i)
# 抠图
results = seg.segment(paths=imgs[:-1], visualization=True ,output_dir=out)